/*实现异步缓冲区*/
#ifndef __M_BUFF_H__
#define __M_BUFF_H__

#include <string>

#include "util.hpp"
#include <vector>
#include <cassert>

namespace bitlog {
#define DEBUG_BUFFER_SIZE (1024 * 1024 * 1)
#define THRESHOLD_BUFFER_SIZE (1024 * 1024 * 8)
#define INCREMENT_BUFFER_SIZE (1024 * 1024 * 1)
    class Buffer {
    public:
        Buffer() : _buffer(DEBUG_BUFFER_SIZE), _reader_idx(0), _writer_idx(0) {}

        // 往缓冲区写入数据
        void push(const char* data, size_t len) {
            // 1.考虑空间不够则扩容
            ensureEnoughSize(len);
            // 2.将数据拷贝到缓冲区中
            std::copy(data, data + len, &_buffer[_writer_idx]);
            // 3.将当前写入位置移动到可写的位置
            moveWriterIdx(len);
        }

        // 获取可读数据的起始指针
        const char* begin() {
            return &_buffer[_reader_idx];
        }

        // 获取可写数据的长度
        size_t writeAbleSize() {
            // 对于扩容思路来说，不存在可写空间大小，因为总是可写
            // 因此这个接口仅仅针对固定大小缓冲区提供
            return _buffer.size() - _writer_idx;
        }
        // 获取可读数据的长度
        size_t readAbleSize() {
            // 因为当前实现的缓冲区设计思想是双缓冲区，处理完就交换，所以不存在空间循环使用
            return _writer_idx - _reader_idx;
        }
        // 重置重写位置，重置缓冲区
        void reset() {
            _reader_idx = _writer_idx = 0; // 缓冲区所有空间都是空闲的
        }
        // 交换缓冲区
        void swap(Buffer& otherBufer) {
            _buffer.swap(otherBufer._buffer);
            std::swap(_reader_idx, otherBufer._reader_idx);
            std::swap(_writer_idx, otherBufer._writer_idx);
        }
        // 判断缓冲区是否为空
        bool empty() {
            return _reader_idx == _writer_idx;
        }

        // 对读指针进行向后移偏移操作
        void moveReaderIdx(size_t len) {
            assert(len <= readAbleSize());
            _reader_idx += len;
        }

    private:
        // 对空间进行扩容
        void ensureEnoughSize(size_t len) {
            if (len <= writeAbleSize())
                return; // 如果可写的空间足够，则不需要扩容
            size_t new_size = 0;
            if (_buffer.size() < THRESHOLD_BUFFER_SIZE) {
                new_size = _buffer.size() * 2 + len; // 小于阈值，则扩容为原来的两倍
            }
            else {
                new_size = _buffer.size() + INCREMENT_BUFFER_SIZE + len; // 大于阈值，则线性增长
            }
            _buffer.resize(new_size);
        }

        // 对写指针进行向后移偏移操作
        void moveWriterIdx(size_t len) {
            assert((len + _writer_idx) <= _buffer.size());
            _writer_idx += len;
        }

    private:
        std::vector<char> _buffer; // 缓冲区
        size_t _reader_idx;        // 当前可读数据的指针--本质是下标
        size_t _writer_idx;        // 当前可写的指针
    };
}

#endif